from ultralytics import YOLO


def train():
    model = YOLO('.\\weight\\pretrained\\yolov8n.pt')  # pass any model type
    model.export()
    results = model.train(epochs=300, device=["0"], imgsz=720)
